#ifndef MY_AES_H
#define MY_AES_H

#ifdef __cplusplus
extern "C"
{
#endif

# define AES_MAXNR 14
# define AES_BLOCK_SIZE 16

#define AES_ENCRYPT     1
#define AES_DECRYPT     0

# define MAXKC   (256/32)
# define MAXKB   (256/8)
# define MAXNR   14

/* This should be a hidden type, but EVP requires that the size be known */
struct aes_key_st {
    unsigned int rd_key[4 * (AES_MAXNR + 1)];
    int rounds;
};
typedef struct aes_key_st AES_KEY;

int AES_set_encrypt_key(const unsigned char *userKey, const int bits,
                        AES_KEY *key);
int AES_set_decrypt_key(const unsigned char *userKey, const int bits,
                        AES_KEY *key);

void AES_cbc_encrypt(const unsigned char *in, unsigned char *out,
                        unsigned int length, const AES_KEY *key,
                        unsigned char *ivec, const int enc);

typedef void (*block128_f)(const unsigned char *in, unsigned char *out, const AES_KEY *key);

void AES_decrypt(const unsigned char *in, unsigned char *out,
                 const AES_KEY *key);
void AES_encrypt(const unsigned char *in, unsigned char *out,
                 const AES_KEY *key);

#ifdef __cplusplus
}
#endif

#endif

